import cv2

HAARCASCADE_FRONTALFACE = './cascades/haarcascade_frontalface_default.xml'
HAARCASCADE_EYE = './cascades/haarcascade_eye.xml'


class HumanDetect(object):
    def __init__(self, cascade):
        self.face_cascade = cv2.CascadeClassifier(cascade)

    def detect(self, image):
        return self.face_cascade.detectMultiScale(image, 1.1, 10)


class FaceDetect(HumanDetect):
    def __init__(self):
        HumanDetect.__init__(self, HAARCASCADE_FRONTALFACE)


class EyeDetect(HumanDetect):
    def __init__(self):
        HumanDetect.__init__(self, HAARCASCADE_EYE)
